package elementaryAlgorithm;

import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

/**
 * 94. 二叉树的中序遍历
 * @描述 给定一个二叉树的根节点 root ，返回它的 中序 遍历。
 *
 * @示例1：
 *
 *    1
 *      \
 *       2
 *      /
 *    3
 * 输入：root = [1,null,2,3]
 * 输出：[1,3,2]
 *
 * @示例2：
 *
 * 输入：root = []
 * 输出：[]
 *
 * @示例3：
 *
 * 输入：root = [1]
 * 输出：[1]
 *
 * @示例4：
 *
 *       1
 *      /
 *    2
 * 输入：root = [1,2]
 * 输出：[2,1]
 *
 * @示例5：
 *
 *      1
 *       \
 *        2
 *
 * 输入：root = [1,null,2]
 * 输出：[1,2]
 *
 *
 * 提示：
 *
 * 树中节点数目在范围 [0, 100] 内
 * -100 <= Node.val <= 100
 *
 *
 * 进阶: 递归算法很简单，你可以通过迭代算法完成吗？
 * */
public class InOorderTraversalOfBinaryTree {

    List<Integer> result = new ArrayList<>();

    /**
     * 二叉树中序遍历，递归解法
     * */
    public List<Integer> inorderTraversal(TreeNode root) {
        if (root == null){
            return result;
        }
        inorderTraversal(root.left);
        result.add(root.val);
        inorderTraversal(root.right);
        return result;
    }

    /**
     * 二叉树中序遍历，循环解法
     * */
    public List<Integer> inorderTraversal2(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        TreeNode cur = root;
        while (cur != null || !stack.isEmpty()) {
            if (cur != null) {
                stack.push(cur);
                cur = cur.left;
            } else {
                cur = stack.pop();
                list.add(cur.val);
                cur = cur.right;
            }
        }
        return list;
    }

    public static void main(String[] args) {
        TreeNode root = new TreeNode(1);
        root.right = new TreeNode(2);
        root.right.left = new TreeNode(3);
        InOorderTraversalOfBinaryTree inOorderTraversalOfBinaryTree = new InOorderTraversalOfBinaryTree();
        List<Integer> integers = inOorderTraversalOfBinaryTree.inorderTraversal2(root);
        System.out.println(integers);
    }
}

//Definition for a binary tree node.
